package com.ruoyi.system.service.impl;

import cn.hutool.core.util.StrUtil;
import com.ruoyi.common.constant.UserConstants;
import com.ruoyi.common.core.domain.PageQuery;
import com.ruoyi.common.core.domain.entity.SysPost;
import com.ruoyi.common.core.domain.entity.SysPostFetcher;
import com.ruoyi.common.core.domain.entity.SysPostTable;
import com.ruoyi.common.core.domain.entity.dto.SysPostPageInput;
import com.ruoyi.common.core.domain.entity.dto.SysPostPageView;
import com.ruoyi.common.core.domain.entity.dto.SysPostSaveInput;
import com.ruoyi.common.core.domain.entity.dto.SysPostUpdateInput;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.system.service.ISysPostService;
import lombok.RequiredArgsConstructor;
import org.babyfish.jimmer.Page;
import org.babyfish.jimmer.sql.JSqlClient;
import org.babyfish.jimmer.sql.ast.mutation.SaveMode;
import org.springframework.stereotype.Service;

import java.util.Arrays;
import java.util.List;

/**
 * 岗位信息 服务层处理
 *
 * @author ruoyi
 */
@Service
@RequiredArgsConstructor
public class SysPostServiceImpl implements ISysPostService {
    private final JSqlClient sqlClient;

    /**
     * 查询岗位信息集合
     *
     * @param post 岗位信息
     * @return 岗位信息集合
     */
    @Override
    public List<SysPostPageView> selectPostList(SysPostPageInput post) {
        SysPostTable postTable = SysPostTable.$;
        return sqlClient.createQuery(postTable)
                .where(
                        postTable.postCode().ilikeIf(post.getPostCode()),
                        postTable.postName().ilikeIf(post.getPostName()),
                        postTable.status().eqIf(post.getStatus())
                )
                .select(postTable.fetch(SysPostPageView.class))
                .execute();
    }

    /**
     * 查询所有岗位
     *
     * @return 岗位列表
     */
    @Override
    public List<SysPost> selectPostAll() {
        SysPostTable table = SysPostTable.$;
        return sqlClient.createQuery(table)
                .select(table.fetch(SysPostFetcher.$.allScalarFields()))
                .execute();
    }

    /**
     * 通过岗位ID查询岗位信息
     *
     * @param postId 岗位ID
     * @return 角色对象信息
     */
    @Override
    public SysPost selectPostById(String postId) {
        return sqlClient.findById(SysPost.class, postId);
    }
    /**
     * 根据用户ID获取岗位选择框列表
     *
     * @param userId 用户ID
     * @return 选中岗位ID列表
     */
   /* @Override
    public List<Long> selectPostListByUserId(Long userId) {
        return postMapper.selectPostListByUserId(userId);
    }*/

    /**
     * 校验岗位名称是否唯一
     *
     * @param postId   岗位信息
     * @param postName 岗位信息
     * @return 结果
     */
    @Override
    public boolean checkPostNameUnique(String postId, String postName) {
        SysPostTable postTable = SysPostTable.$;
        SysPost info = sqlClient.createQuery(postTable)
                .where(postTable.postName().eq(postName))
                .select(postTable.fetch(SysPostFetcher.$.allScalarFields()))
                .fetchOneOrNull();
        if (StringUtils.isNotNull(info) && info.postId() != postId) {
            return UserConstants.NOT_UNIQUE;
        }
        return UserConstants.UNIQUE;
    }

    /**
     * 校验岗位编码是否唯一
     *
     * @param postId   岗位信息
     * @param postCode 岗位信息
     * @return 结果
     */
    @Override
    public boolean checkPostCodeUnique(String postId, String postCode) {
        SysPostTable postTable = SysPostTable.$;
        SysPost info = sqlClient.createQuery(postTable)
                .where(postTable.postCode().eq(postCode))
                .select(postTable.fetch(SysPostFetcher.$.allScalarFields()))
                .fetchOneOrNull();
        if (StringUtils.isNotNull(info) && info.postId() != postId) {
            return UserConstants.NOT_UNIQUE;
        }
        return UserConstants.UNIQUE;
    }

    /**
     * 通过岗位ID查询岗位使用数量
     *
     * @param postId 岗位ID
     * @return 结果
     */
    @Override
    public int countUserPostById(String postId) {
        SysPostTable postTable = SysPostTable.$;
        SysPost sysPost = sqlClient.createQuery(postTable)
                .where(postTable.postId().eq(postId))
                .select(postTable.fetch(SysPostFetcher.$.users()))
                .fetchOne();
        return sysPost.users().size();
    }
    /**
     * 删除岗位信息
     *
     * @param postId 岗位ID
     * @return 结果
     */
    /*@Override
    public int deletePostById(Long postId) {
        return postMapper.deletePostById(postId);
    }*/

    /**
     * 批量删除岗位信息
     *
     * @param postIds 需要删除的岗位ID
     * @return 结果
     */
    @Override
    public int deletePostByIds(String[] postIds) {
        for (String postId : postIds) {
            SysPost post = selectPostById(postId);
            if (countUserPostById(postId) > 0) {
                throw new ServiceException(String.format("%1$s已分配,不能删除", post.postName()));
            }
        }

        SysPostTable table = SysPostTable.$;
        return sqlClient.createDelete(table)
                .where(table.postId().in(Arrays.asList(postIds)))
                .execute();
    }

    /**
     * 新增保存岗位信息
     *
     * @param post 岗位信息
     * @return 结果
     */
    @Override
    public int insertPost(SysPostSaveInput post) {
        return sqlClient.saveCommand(post)
                .setMode(SaveMode.INSERT_ONLY)
                .execute()
                .getTotalAffectedRowCount();
    }

    /**
     * 修改保存岗位信息
     *
     * @param post 岗位信息
     * @return 结果
     */
    @Override
    public int updatePost(SysPostUpdateInput post) {
        return sqlClient.saveCommand(post).setMode(SaveMode.UPDATE_ONLY).execute().getTotalAffectedRowCount();
    }

    @Override
    public Page<SysPostPageView> selectPostPage(SysPostPageInput post, PageQuery pageQuery) {
        SysPostTable postTable = SysPostTable.$;
        return sqlClient.createQuery(postTable)
                .where(
                        postTable.postCode().ilikeIf(post.getPostCode()),
                        postTable.postName().ilikeIf(post.getPostName()),
                        postTable.status().eqIf(post.getStatus())
                )
                .select(postTable.fetch(SysPostPageView.class))
                .fetchPage(pageQuery.getPageNum(), pageQuery.getPageSize());
    }
}
